Iterating over dictionary keys with a function